Cython: translation.pyx

Yellow lines hint at Python interaction.
Click on a line that starts with a "+" to see the C code that Cython generated for it.

+01: # cython: infer_types=True, wraparound=False, nonecheck=False, boundscheck=False, cdivision=True, language_level=3, profile=False, autogen_pxd=True

  __pyx_t_7 = __Pyx_PyDict_NewPresized(0); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 1, __pyx_L1_error)
  __Pyx_GOTREF(__pyx_t_7);
  if (PyDict_SetItem(__pyx_d, __pyx_n_s_test, __pyx_t_7) < 0) __PYX_ERR(0, 1, __pyx_L1_error)
  __Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;

02:

+03: import numpy as np

  __pyx_t_7 = __Pyx_ImportDottedModule(__pyx_n_s_numpy, NULL); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 3, __pyx_L1_error)
  __Pyx_GOTREF(__pyx_t_7);
  if (PyDict_SetItem(__pyx_d, __pyx_n_s_np, __pyx_t_7) < 0) __PYX_ERR(0, 3, __pyx_L1_error)
  __Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;

 04: cimport numpy as np

05:

 06: from cython.parallel import prange

07:

 08: from .interpolation_catmull_rom cimport Interpolator

09:

10:

+11: def translate_array(float[:, :, :] img_arr, float[:, :] drift_t):

/* Python wrapper */
static PyObject *__pyx_pw_7nanopyx_4core_9transform_11translation_1translate_array(PyObject *__pyx_self, 
#if CYTHON_METH_FASTCALL
PyObject *const *__pyx_args, Py_ssize_t __pyx_nargs, PyObject *__pyx_kwds
#else
PyObject *__pyx_args, PyObject *__pyx_kwds
#endif
); /*proto*/
PyDoc_STRVAR(__pyx_doc_7nanopyx_4core_9transform_11translation_translate_array, "\n    Translate an array of images using the drift data.\n    ");
static PyMethodDef __pyx_mdef_7nanopyx_4core_9transform_11translation_1translate_array = {"translate_array", (PyCFunction)(void*)(__Pyx_PyCFunction_FastCallWithKeywords)__pyx_pw_7nanopyx_4core_9transform_11translation_1translate_array, __Pyx_METH_FASTCALL|METH_KEYWORDS, __pyx_doc_7nanopyx_4core_9transform_11translation_translate_array};
static PyObject *__pyx_pw_7nanopyx_4core_9transform_11translation_1translate_array(PyObject *__pyx_self, 
#if CYTHON_METH_FASTCALL
PyObject *const *__pyx_args, Py_ssize_t __pyx_nargs, PyObject *__pyx_kwds
#else
PyObject *__pyx_args, PyObject *__pyx_kwds
#endif
) {
  __Pyx_memviewslice __pyx_v_img_arr = { 0, 0, { 0 }, { 0 }, { 0 } };
  __Pyx_memviewslice __pyx_v_drift_t = { 0, 0, { 0 }, { 0 }, { 0 } };
  #if !CYTHON_METH_FASTCALL
  CYTHON_UNUSED Py_ssize_t __pyx_nargs;
  #endif
  CYTHON_UNUSED PyObject *const *__pyx_kwvalues;
  PyObject *__pyx_r = 0;
  __Pyx_RefNannyDeclarations
  __Pyx_RefNannySetupContext("translate_array (wrapper)", 0);
  #if !CYTHON_METH_FASTCALL
  #if CYTHON_ASSUME_SAFE_MACROS
  __pyx_nargs = PyTuple_GET_SIZE(__pyx_args);
  #else
  __pyx_nargs = PyTuple_Size(__pyx_args);
  if (unlikely((__pyx_nargs < 0))) __PYX_ERR(0, 11, __pyx_L3_error)
  #endif
  #endif
  __pyx_kwvalues = __Pyx_KwValues_FASTCALL(__pyx_args, __pyx_nargs);
  {
    PyObject **__pyx_pyargnames[] = {&__pyx_n_s_img_arr,&__pyx_n_s_drift_t,0};
  PyObject* values[2] = {0,0};
    if (__pyx_kwds) {
      Py_ssize_t kw_args;
      switch (__pyx_nargs) {
        case  2: values[1] = __Pyx_Arg_FASTCALL(__pyx_args, 1);
        CYTHON_FALLTHROUGH;
        case  1: values[0] = __Pyx_Arg_FASTCALL(__pyx_args, 0);
        CYTHON_FALLTHROUGH;
        case  0: break;
        default: goto __pyx_L5_argtuple_error;
      }
      kw_args = __Pyx_NumKwargs_FASTCALL(__pyx_kwds);
      switch (__pyx_nargs) {
        case  0:
        if (likely((values[0] = __Pyx_GetKwValue_FASTCALL(__pyx_kwds, __pyx_kwvalues, __pyx_n_s_img_arr)) != 0)) {
          (void)__Pyx_Arg_NewRef_FASTCALL(values[0]);
          kw_args--;
        }
        else if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 11, __pyx_L3_error)
        else goto __pyx_L5_argtuple_error;
        CYTHON_FALLTHROUGH;
        case  1:
        if (likely((values[1] = __Pyx_GetKwValue_FASTCALL(__pyx_kwds, __pyx_kwvalues, __pyx_n_s_drift_t)) != 0)) {
          (void)__Pyx_Arg_NewRef_FASTCALL(values[1]);
          kw_args--;
        }
        else if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 11, __pyx_L3_error)
        else {
          __Pyx_RaiseArgtupleInvalid("translate_array", 1, 2, 2, 1); __PYX_ERR(0, 11, __pyx_L3_error)
        }
      }
      if (unlikely(kw_args > 0)) {
        const Py_ssize_t kwd_pos_args = __pyx_nargs;
        if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_kwvalues, __pyx_pyargnames, 0, values + 0, kwd_pos_args, "translate_array") < 0)) __PYX_ERR(0, 11, __pyx_L3_error)
      }
    } else if (unlikely(__pyx_nargs != 2)) {
      goto __pyx_L5_argtuple_error;
    } else {
      values[0] = __Pyx_Arg_FASTCALL(__pyx_args, 0);
      values[1] = __Pyx_Arg_FASTCALL(__pyx_args, 1);
    }
    __pyx_v_img_arr = __Pyx_PyObject_to_MemoryviewSlice_dsdsds_float(values[0], PyBUF_WRITABLE); if (unlikely(!__pyx_v_img_arr.memview)) __PYX_ERR(0, 11, __pyx_L3_error)
    __pyx_v_drift_t = __Pyx_PyObject_to_MemoryviewSlice_dsds_float(values[1], PyBUF_WRITABLE); if (unlikely(!__pyx_v_drift_t.memview)) __PYX_ERR(0, 11, __pyx_L3_error)
  }
  goto __pyx_L4_argument_unpacking_done;
  __pyx_L5_argtuple_error:;
  __Pyx_RaiseArgtupleInvalid("translate_array", 1, 2, 2, __pyx_nargs); __PYX_ERR(0, 11, __pyx_L3_error)
  goto __pyx_L3_error;
  __pyx_L3_error:;
  {
    Py_ssize_t __pyx_temp;
    for (__pyx_temp=0; __pyx_temp < (Py_ssize_t)(sizeof(values)/sizeof(values[0])); ++__pyx_temp) {
      __Pyx_Arg_XDECREF_FASTCALL(values[__pyx_temp]);
    }
  }
  __PYX_XCLEAR_MEMVIEW(&__pyx_v_img_arr, 1);
  __PYX_XCLEAR_MEMVIEW(&__pyx_v_drift_t, 1);
  __Pyx_AddTraceback("nanopyx.core.transform.translation.translate_array", __pyx_clineno, __pyx_lineno, __pyx_filename);
  __Pyx_RefNannyFinishContext();
  return NULL;
  __pyx_L4_argument_unpacking_done:;
  __pyx_r = __pyx_pf_7nanopyx_4core_9transform_11translation_translate_array(__pyx_self, __pyx_v_img_arr, __pyx_v_drift_t);
  int __pyx_lineno = 0;
  const char *__pyx_filename = NULL;
  int __pyx_clineno = 0;

  /* function exit code */
  __PYX_XCLEAR_MEMVIEW(&__pyx_v_img_arr, 1);
  __PYX_XCLEAR_MEMVIEW(&__pyx_v_drift_t, 1);
  {
    Py_ssize_t __pyx_temp;
    for (__pyx_temp=0; __pyx_temp < (Py_ssize_t)(sizeof(values)/sizeof(values[0])); ++__pyx_temp) {
      __Pyx_Arg_XDECREF_FASTCALL(values[__pyx_temp]);
    }
  }
  __Pyx_RefNannyFinishContext();
  return __pyx_r;
}

static PyObject *__pyx_pf_7nanopyx_4core_9transform_11translation_translate_array(CYTHON_UNUSED PyObject *__pyx_self, __Pyx_memviewslice __pyx_v_img_arr, __Pyx_memviewslice __pyx_v_drift_t) {
  int __pyx_v_n_slices;
  float __pyx_v_drift_x;
  float __pyx_v_drift_y;
  int __pyx_v_i;
  PyObject *__pyx_r = NULL;
  __Pyx_RefNannyDeclarations
  __Pyx_RefNannySetupContext("translate_array", 0);
/* … */
  /* function exit code */
  __pyx_L1_error:;
  __PYX_XCLEAR_MEMVIEW(&__pyx_t_6, 1);
  __Pyx_XDECREF(__pyx_t_7);
  __Pyx_XDECREF(__pyx_t_8);
  __PYX_XCLEAR_MEMVIEW(&__pyx_t_9, 1);
  __Pyx_AddTraceback("nanopyx.core.transform.translation.translate_array", __pyx_clineno, __pyx_lineno, __pyx_filename);
  __pyx_r = NULL;
  __pyx_L0:;
  __Pyx_XGIVEREF(__pyx_r);
  __Pyx_RefNannyFinishContext();
  return __pyx_r;
}
/* … */
  __pyx_tuple__22 = PyTuple_Pack(6, __pyx_n_s_img_arr, __pyx_n_s_drift_t, __pyx_n_s_n_slices, __pyx_n_s_drift_x, __pyx_n_s_drift_y, __pyx_n_s_i); if (unlikely(!__pyx_tuple__22)) __PYX_ERR(0, 11, __pyx_L1_error)
  __Pyx_GOTREF(__pyx_tuple__22);
  __Pyx_GIVEREF(__pyx_tuple__22);
/* … */
  __pyx_t_7 = __Pyx_CyFunction_New(&__pyx_mdef_7nanopyx_4core_9transform_11translation_1translate_array, 0, __pyx_n_s_translate_array, NULL, __pyx_n_s_nanopyx_core_transform_translati, __pyx_d, ((PyObject *)__pyx_codeobj__23)); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 11, __pyx_L1_error)
  __Pyx_GOTREF(__pyx_t_7);
  if (PyDict_SetItem(__pyx_d, __pyx_n_s_translate_array, __pyx_t_7) < 0) __PYX_ERR(0, 11, __pyx_L1_error)
  __Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;

 12:     """

 13:     Translate an array of images using the drift data.

 14:     """

15:

+16:     cdef int n_slices = img_arr.shape[0]

  __pyx_v_n_slices = (__pyx_v_img_arr.shape[0]);

 17:     cdef float drift_x, drift_y

18:

+19:     for i in range(n_slices):

  __pyx_t_1 = __pyx_v_n_slices;
  __pyx_t_2 = __pyx_t_1;
  for (__pyx_t_3 = 0; __pyx_t_3 < __pyx_t_2; __pyx_t_3+=1) {
    __pyx_v_i = __pyx_t_3;

+20:         drift_x = drift_t[i][1]

    __pyx_t_4 = __pyx_v_i;
    __pyx_t_5 = 1;
    __pyx_v_drift_x = (*((float *) ( /* dim=1 */ (( /* dim=0 */ (__pyx_v_drift_t.data + __pyx_t_4 * __pyx_v_drift_t.strides[0]) ) + __pyx_t_5 * __pyx_v_drift_t.strides[1]) )));

+21:         drift_y = drift_t[i][2]

    __pyx_t_5 = __pyx_v_i;
    __pyx_t_4 = 2;
    __pyx_v_drift_y = (*((float *) ( /* dim=1 */ (( /* dim=0 */ (__pyx_v_drift_t.data + __pyx_t_5 * __pyx_v_drift_t.strides[0]) ) + __pyx_t_4 * __pyx_v_drift_t.strides[1]) )));

+22:         img_arr[i,  :, :] = Interpolator(img_arr[i])._shift(drift_y, drift_x)

    __pyx_t_6.data = __pyx_v_img_arr.data;
    __pyx_t_6.memview = __pyx_v_img_arr.memview;
    __PYX_INC_MEMVIEW(&__pyx_t_6, 1);
    {
    Py_ssize_t __pyx_tmp_idx = __pyx_v_i;
    Py_ssize_t __pyx_tmp_stride = __pyx_v_img_arr.strides[0];
        __pyx_t_6.data += __pyx_tmp_idx * __pyx_tmp_stride;
}

__pyx_t_6.shape[0] = __pyx_v_img_arr.shape[1];
__pyx_t_6.strides[0] = __pyx_v_img_arr.strides[1];
    __pyx_t_6.suboffsets[0] = -1;

__pyx_t_6.shape[1] = __pyx_v_img_arr.shape[2];
__pyx_t_6.strides[1] = __pyx_v_img_arr.strides[2];
    __pyx_t_6.suboffsets[1] = -1;

__pyx_t_7 = __pyx_memoryview_fromslice(__pyx_t_6, 2, (PyObject *(*)(char *)) __pyx_memview_get_float, (int (*)(char *, PyObject *)) __pyx_memview_set_float, 0);; if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 22, __pyx_L1_error)
    __Pyx_GOTREF(__pyx_t_7);
    __PYX_XCLEAR_MEMVIEW(&__pyx_t_6, 1);
    __pyx_t_6.memview = NULL; __pyx_t_6.data = NULL;
    __pyx_t_8 = __Pyx_PyObject_CallOneArg(((PyObject *)__pyx_ptype_7nanopyx_4core_9transform_25interpolation_catmull_rom_Interpolator), __pyx_t_7); if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 22, __pyx_L1_error)
    __Pyx_GOTREF(__pyx_t_8);
    __Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
    __pyx_t_6 = ((struct __pyx_vtabstruct_7nanopyx_4core_9transform_25interpolation_catmull_rom_Interpolator *)((struct __pyx_obj_7nanopyx_4core_9transform_25interpolation_catmull_rom_Interpolator *)__pyx_t_8)->__pyx_base.__pyx_vtab)->__pyx_base._shift(((struct __pyx_obj_7nanopyx_4core_9transform_30interpolation_nearest_neighbor_Interpolator *)__pyx_t_8), __pyx_v_drift_y, __pyx_v_drift_x); if (unlikely(!__pyx_t_6.memview)) __PYX_ERR(0, 22, __pyx_L1_error)
    __Pyx_DECREF(__pyx_t_8); __pyx_t_8 = 0;
    __pyx_t_9.data = __pyx_v_img_arr.data;
    __pyx_t_9.memview = __pyx_v_img_arr.memview;
    __PYX_INC_MEMVIEW(&__pyx_t_9, 1);
    {
    Py_ssize_t __pyx_tmp_idx = __pyx_v_i;
    Py_ssize_t __pyx_tmp_stride = __pyx_v_img_arr.strides[0];
        __pyx_t_9.data += __pyx_tmp_idx * __pyx_tmp_stride;
}

__pyx_t_9.shape[0] = __pyx_v_img_arr.shape[1];
__pyx_t_9.strides[0] = __pyx_v_img_arr.strides[1];
    __pyx_t_9.suboffsets[0] = -1;

__pyx_t_9.shape[1] = __pyx_v_img_arr.shape[2];
__pyx_t_9.strides[1] = __pyx_v_img_arr.strides[2];
    __pyx_t_9.suboffsets[1] = -1;

if (unlikely((__pyx_memoryview_copy_contents(__pyx_t_6, __pyx_t_9, 2, 2, 0) < 0))) __PYX_ERR(0, 22, __pyx_L1_error)
    __PYX_XCLEAR_MEMVIEW(&__pyx_t_9, 1);
    __pyx_t_9.memview = NULL; __pyx_t_9.data = NULL;
    __PYX_XCLEAR_MEMVIEW(&__pyx_t_6, 1);
    __pyx_t_6.memview = NULL; __pyx_t_6.data = NULL;
  }

23:

+24:     return img_arr

  __Pyx_XDECREF(__pyx_r);
  __pyx_t_8 = __pyx_memoryview_fromslice(__pyx_v_img_arr, 3, (PyObject *(*)(char *)) __pyx_memview_get_float, (int (*)(char *, PyObject *)) __pyx_memview_set_float, 0);; if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 24, __pyx_L1_error)
  __Pyx_GOTREF(__pyx_t_8);
  __pyx_r = __pyx_t_8;
  __pyx_t_8 = 0;
  goto __pyx_L0;